"""
功能：绘制信号时频图谱
技术：Matplotlib可视化
比喻：把振动信号变成音乐波形图
"""
import matplotlib.pyplot as plt

# 设置中文字体（替换为系统已安装的字体）
plt.rcParams['font.sans-serif'] = ['SimHei', 'Noto Sans CJK JP']  # 优先使用SimHei
plt.rcParams['axes.unicode_minus'] = False  # 解决负号显示问题


def paint_signals(t, healthy, damaged, save_path='report.png'):
    """
    参数说明：
    t        : 时间轴 -> 画布长度
    healthy  : 健康信号 -> 基准色
    damaged  : 损伤信号 -> 对比色
    save_path: 保存路径 -> 画框位置
    """
    plt.figure(figsize=(10, 6))

    # 绘制双信号对比（健康vs损伤）
    plt.plot(t, healthy, label='健康信号', alpha=0.7)
    plt.plot(t, damaged, label='损伤信号', color='red', linestyle='--')

    # 装饰画布
    plt.title('结构心电图对比')
    plt.xlabel('时间(s)');
    plt.ylabel('振幅');
    plt.legend()
    plt.grid(alpha=0.3)
    plt.savefig(save_path)
    #plt.close()  # 关闭图像，避免内存泄漏